import platform


img_dir_mac = "/Users/leslie/Desktop/dataset"


img_dir_linux = "/home/leslie/project/ViT/dataset_shucai"
image_valid_path = 'images_valid_shucai.npy'


def  get_dataset_dir():
    if platform.system().lower() == 'darwin':
        return img_dir_mac
    if platform.system().lower() == 'linux':
        return img_dir_linux

def a(b):
    b=3

if __name__ == '__main__':

    b = 2
    a(b)
    print(b)

